from django.contrib import admin

from erp.models import Choice, Question


class ChoiceInline(admin.TabularInline):
    model = Choice
    extra = 3


class ChoiceAdmin(admin.ModelAdmin):
    list_display = ('choice_text', 'votes')
    list_filter = ['votes']
    search_fields = ['choice_text', 'votes']
    # fieldsets = [
    #     (None, {'fields': ['choice_text']}),
    #     ('Votes', {'fields': ['votes'], 'classes': ['collapse']}),
    # ]


class QuestionAdmin(admin.ModelAdmin):
    list_display = ('question_text', 'pub_date', 'was_published_recently')
    list_filter = ['pub_date']
    search_fields = ['question_text', 'pub_date']
    fieldsets = [
        (None, {'fields': ['question_text']}),
        ('Date information', {'fields': ['pub_date'], 'classes': ['collapse']}),
    ]
    inlines = [ChoiceInline]


admin.AdminSite.site_header = ''
admin.site.register(Question, QuestionAdmin)
admin.site.register(Choice, ChoiceAdmin)
